In [1]:
%run train-cGAN
# %run unit_test
self.device is cuda:0
64
Epoch: 0, Itration: 0
loss_G_GAN = 5.2222394943237305, loss_G_L1 = 0.42152929306030273 * 5.0
loss_D = 0.6994704008102417
Epoch: 76, Itration: 1000
loss_G_GAN = 1.7559757232666016, loss_G_L1 = 0.05914055183529854 * 5.0
loss_D = 0.32040828466415405
Epoch: 153, Itration: 2000
loss_G_GAN = 2.0574169158935547, loss_G_L1 = 0.05481714755296707 * 5.0
loss_D = 0.2651404142379761
save model to ./checkpoints/pix2food.pkl at iteration = 3000
Epoch: 230, Itration: 3000
loss_G_GAN = 1.2604695558547974, loss_G_L1 = 0.04347262904047966 * 5.0
loss_D = 0.3395516872406006
Finally save model to ./checkpoints/pix2food.pkl



 # --- Validation Set --- # 



In [3]:
# %run train-cGAN
In [2]:
# %run train
In [ ]:
 
In [ ]: